#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2021/11/28 15:05
# @Author : xjp
# @Site : 
# @File : plt.py
# @Software: PyCharm
from matplotlib import pyplot, pyplot as plt


def plt_loss(list):
    pyplot.plot(list)
    return pyplot.show()
def GAN_loss(list1,list2):
    plt.figure(figsize=(20, 10))
    plt.plot(list1, color='blue', marker='o', linestyle="--", label="Gen")
    plt.plot(list2, color='red', marker='+', linestyle="-", label="Dis")
    plt.title("LOSS")
    plt.legend(loc="upper right")
    return pyplot.show()

